package com.ilxqx.generator.data;

import com.ilxqx.generator.parser.Table;

import java.util.Map;

/**
 * 主键处理
 * @author venus
 */
@ProcessorOrder(1)
public class PrimaryKeyDataModelProcessor implements DataModelProcessor {

	/**
	 * 数据模型处理
	 *
	 * @param dataModel 数据模型(其中table为表对象，里面包含你想要的信息)
	 */
	@Override
	public void process(Map<String, Object> dataModel) {
		Table table = (Table) dataModel.get("table");
		table.getFields().forEach(tableField -> {
			if (tableField.isPrimaryKey()) {
				dataModel.put("primary", tableField);
			}
		});
	}

}
